package com.cdqidi.core.weixinmanager.service;

import com.baomidou.mybatisplus.core.conditions.query.QueryWrapper;
import com.baomidou.mybatisplus.core.metadata.IPage;
import com.baomidou.mybatisplus.extension.plugins.pagination.Page;
import com.cdqidi.constant.ConstantDto;
import com.cdqidi.core.weixinmanager.domain.MiddleUserDTO;
import com.cdqidi.core.weixinmanager.mapper.MiddleUserMapper;
import com.cdqidi.core.weixinmanager.model.MiddleUser;
import com.cdqidi.db.BaseService;
import com.cdqidi.dto.CacheKey;
import com.cdqidi.excel.ExcelExport;
import com.cdqidi.excel.util.ExportPage;
import com.cdqidi.exception.ApiException;
import com.cdqidi.util.RedisTemplateUtil;
import com.qd.common.sys.constant.Constants;
import lombok.RequiredArgsConstructor;
import lombok.extern.slf4j.Slf4j;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;

import javax.annotation.Resource;
import javax.servlet.http.HttpServletResponse;
import java.io.IOException;
import java.util.Collections;
import java.util.List;
import java.util.Optional;

import static java.util.stream.Collectors.toList;

/**
 * 人员和微信用户
 *
 * @author sjk
 */
@RequiredArgsConstructor
@Service
@Slf4j
public class MiddleUserService extends BaseService<MiddleUserMapper, MiddleUser> {
    private static final String PREFIX_ID = "sys_wx_middle_user_rId_{0}";
    private static final String PREFIX_UNIQUE_SYS_WX_MID = "sys_wx_middle_user_userId_{0}_appId_{1}";
    private static final String PREFIX_OPENID_APP_ID = "sys_wx_middle_user_openid_{0}_appId_{1}";
    private static final String PREFIX_LIST = "sys_wx_middle_user_userId_{0}";

    private final RedisTemplateUtil<MiddleUser> redisTemplateUtil;

    private final OrgWxConfigService orgWxConfigService;
    @Resource
    private HttpServletResponse response;

    /**
     * 保存
     *
     * @param dto 页面对象
     * @return 页面对象
     */
    @Transactional(rollbackFor = Exception.class)
    public MiddleUserDTO save(MiddleUserDTO dto) {
        checkUnique(dto);
        final MiddleUser model = dtoToModel(dto);
        try {
            if (super.save(model)) {
                return modelToDto(model);
            }
            throw new ApiException("添加失败");
        } finally {
            clearCache(model);
            model.freeData();
        }
    }

    /**
     * 修改
     *
     * @param dto 页面对象
     * @return 页面对象
     */
    @Transactional(rollbackFor = Exception.class)
    public MiddleUserDTO update(MiddleUserDTO dto) {
        checkUnique(dto);
        final MiddleUser historyModel = Optional.ofNullable(getById(dto.getRId())).orElseThrow(() -> new ApiException("ID不存在,不能更新"));
        final MiddleUser model = dtoToModel(dto);
        try {
            if (!super.updateById(model)) {
                throw new ApiException("添加失败");
            }
        } finally {
            clearCache(historyModel);
            clearCache(model);
            historyModel.freeData();
            model.freeData();
        }
        return getByIdDto(dto.getRId());
    }


    /**
     * 删除，单个
     *
     * @param rId 主键
     */
    @Transactional(rollbackFor = Exception.class)
    public void removeById(String rId) {
        final MiddleUser model = Optional.ofNullable(getById(rId)).orElseThrow(() -> new ApiException("数据不存在,不能删除"));
        try {
            if (!super.removeById(rId)) {
                throw new ApiException("删除失败");
            }
        } finally {
            clearCache(model);
            model.freeData();
        }
    }

    /**
     * 根据传入对象构建查询参数，判断是否存在
     *
     * @param dto 参数
     * @return 存在为true
     */
    @Transactional(readOnly = true)
    public Boolean isExists(MiddleUserDTO dto) {
        final QueryWrapper<MiddleUser> queryWrapper = createQueryWrapper(dto);
        final boolean exists = super.count(queryWrapper) > 0;
        queryWrapper.clear();
        return exists;
    }

    /**
     * 根据主键判断数据是否存在
     *
     * @param rId 主键
     * @return 存在为true
     */
    @Transactional(readOnly = true)
    public Boolean isExistsById(String rId) {
        boolean exists = false;
        if (StringUtils.hasLength(rId)) {
            final MiddleUser model = getById(rId);
            exists = (null != model);
            if (exists) {
                model.freeData();
            }
        }
        return exists;
    }

    /**
     * 根据主键获取数据
     *
     * @param rId 主键
     * @return 单条记录
     */
    @Transactional(readOnly = true)
    public MiddleUser getById(String rId) {
        final String key = RedisTemplateUtil.getRedisCacheKey(PREFIX_ID, rId);
        return redisTemplateUtil.get(new CacheKey().setKey(key).setTimeout(ConstantDto.REDIS_TIME_OUT_DAY), () -> super.getById(rId), MiddleUser.class);
    }

    /**
     * 根据主键获取数据
     *
     * @param rId 主键
     * @return 单条记录，转换成页面对象
     */
    @Transactional(readOnly = true)
    public MiddleUserDTO getByIdDto(String rId) {
        return Optional.ofNullable(getById(rId)).map(this::modelToDto).orElse(null);
    }

    /**
     * 根据条件查询单条数据
     *
     * @param dto 查询参数
     * @return 单条记录
     */
    @Transactional(readOnly = true)
    public MiddleUser getOne(MiddleUserDTO dto) {
        final QueryWrapper<MiddleUser> queryWrapper = createQueryWrapper(dto);
        MiddleUser model = super.getOne(queryWrapper);
        queryWrapper.clear();
        return model;
    }

    /**
     * 根据条件查询单条数据
     *
     * @param dto 查询参数
     * @return 单条记录，转换成页面对象
     */
    @Transactional(readOnly = true)
    public MiddleUserDTO getOneDto(MiddleUserDTO dto) {
        return Optional.ofNullable(getOne(dto)).map(this::modelToDto).orElse(null);
    }

    /**
     * 根据条件查询列表数据
     *
     * @param dto 查询参数
     * @return 列表数据
     */
    @Transactional(readOnly = true)
    public List<MiddleUser> getList(MiddleUserDTO dto) {
        final QueryWrapper<MiddleUser> queryWrapper = createQueryWrapper(dto);
        final List<MiddleUser> list = super.list(queryWrapper);
        queryWrapper.clear();
        return list;
    }

    @Transactional(readOnly = true)
    public List<MiddleUser> getListByUserId(String userId) {
        String key = RedisTemplateUtil.getRedisCacheKey(PREFIX_LIST, userId);
        return redisTemplateUtil.getList(new CacheKey().setKey(key), () -> {
            QueryWrapper<MiddleUser> queryWrapper = new QueryWrapper<>();
            queryWrapper.eq(Constants.USER_ID, userId);
            List<MiddleUser> list = super.list(queryWrapper);
            queryWrapper.clear();
            return list;
        }, MiddleUser.class);
    }

    /**
     * 根据条件查询列表数据
     *
     * @param dto 查询参数
     * @return 列表数据，转换成页面对象
     */
    @Transactional(readOnly = true)
    public List<MiddleUserDTO> getListDto(MiddleUserDTO dto) {
        return modelToDtoList(getList(dto));
    }

    /**
     * 分页数据
     *
     * @param page 分页对象
     * @param dto  查询参数
     * @return 分页信息
     */
    @Transactional(readOnly = true)
    public IPage<MiddleUserDTO> page(Page<MiddleUser> page, MiddleUserDTO dto) {
        final QueryWrapper<MiddleUser> queryWrapper = createQueryWrapper(dto, "mid.");
        final IPage<MiddleUserDTO> iPage = baseMapper.page(page, queryWrapper);
        queryWrapper.clear();
        return iPage;
    }

    /**
     * 导出Excel
     *
     * @param query 查询条件
     */
    @Transactional(readOnly = true)
    public void export(MiddleUserDTO query) {
        final ExportPage<MiddleUser, MiddleUserDTO> exportPage = new ExportPage<>();
        try {
            final List<MiddleUserDTO> list = exportPage.getAllForPage(page -> page(page, query).getRecords());
            try {
                ExcelExport.downloadExcel(response, MiddleUserDTO.class, () -> list);
            } catch (IOException e) {
                log.error("导出失败", e);
                throw new ApiException("导出失败");
            }
        } finally {
            exportPage.freeData();
        }
    }


    /**
     * 根据唯一性索引获得数据
     * <p>
     * param userId
     * param appId
     *
     * @return 单条数据
     */
    @Transactional(readOnly = true)
    public MiddleUser getUniqueByUserIdAndAppId(String userId, String appId) {
        final QueryWrapper<MiddleUser> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq(Constants.USER_ID, userId);
        queryWrapper.eq(Constants.APPID, appId);
        final String key = RedisTemplateUtil.getRedisCacheKey(PREFIX_UNIQUE_SYS_WX_MID, userId, appId);
        final MiddleUser model = redisTemplateUtil.get(new CacheKey().setKey(key).setTimeout(ConstantDto.REDIS_TIME_OUT_DAY), () -> super.getOne(queryWrapper, false), MiddleUser.class);
        queryWrapper.clear();
        return model;
    }

    /**
     * 根据openid和appId获得用户信息
     *
     * @param openid 微信用户ID
     * @param appId  公众号或者小程序ID
     * @return 用户信息
     */
    @Transactional(readOnly = true)
    public MiddleUser getMiddleUserByOpenidAndAppId(String openid, String appId) {
        final QueryWrapper<MiddleUser> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq(Constants.APPID, appId);
        queryWrapper.eq(Constants.OPENID, openid);
        final String key = RedisTemplateUtil.getRedisCacheKey(PREFIX_OPENID_APP_ID, openid, appId);
        final MiddleUser model = redisTemplateUtil.get(new CacheKey().setKey(key).setTimeout(ConstantDto.REDIS_TIME_OUT_DAY), () -> super.getOne(queryWrapper, false), MiddleUser.class);
        queryWrapper.clear();
        return model;
    }

    /**
     * 验证唯一性索引
     *
     * @param dto 页面对象
     */
    @Transactional(readOnly = true)
    public void checkUnique(MiddleUserDTO dto) {
        MiddleUser model = getUniqueByUserIdAndAppId(dto.getUserId(), dto.getOpenid());
        if (null != model) {
            checkUnique(dto, model);
        }
    }

    /**
     * 验证唯一性索引
     *
     * @param dto   页面对象
     * @param model 根据约束字段查询出来的数据库对象
     */
    private void checkUnique(MiddleUserDTO dto, MiddleUser model) {
        if (!StringUtils.hasLength(dto.getRId())) {
            throw new ApiException("数据已存在，请勿重复添加");
        }
        if (!dto.getRId().equals(model.getRId())) {
            throw new ApiException("数据已存在，请勿重复添加");
        }
    }

    /**
     * 清空缓存
     *
     * @param model 数据库对象
     */
    public void clearCache(MiddleUser model) {
        if (null == model) {
            return;
        }
        redisTemplateUtil.deleteByKey(RedisTemplateUtil.getRedisCacheKey(PREFIX_ID, model.getRId()));
        redisTemplateUtil.deleteByKey(RedisTemplateUtil.getRedisCacheKey(PREFIX_UNIQUE_SYS_WX_MID, model.getUserId(), model.getAppId()));
        redisTemplateUtil.deleteByKey(RedisTemplateUtil.getRedisCacheKey(PREFIX_OPENID_APP_ID, model.getOpenid(), model.getAppId()));
        redisTemplateUtil.deleteByKey(RedisTemplateUtil.getRedisCacheKey(PREFIX_LIST, model.getUserId()));
    }

    /**
     * 数据库对象转页面对象
     *
     * @param model 数据库对象
     * @return 页面对象
     */
    public MiddleUserDTO modelToDto(MiddleUser model) {
        if (null == model) {
            return null;
        }
        final MiddleUserDTO dto = new MiddleUserDTO();
        dto.setRId(model.getRId());
        dto.setUserId(model.getUserId());
        dto.setMobile(model.getMobile());
        dto.setUnionId(model.getUnionId());
        dto.setNickName(model.getNickName());
        dto.setOpenid(model.getOpenid());
        dto.setBindTime(model.getBindTime());
        dto.setHeadImg(model.getHeadImg());
        dto.setAppId(model.getAppId());
        buildDisplay(dto);
        return dto;
    }

    private void buildDisplay(MiddleUserDTO dto) {
        dto.setAppIdDisplay(orgWxConfigService.idToDisplay(dto.getAppId()));
    }

    /**
     * 页面对象转数据库对象
     *
     * @param dto 页面对象
     * @return 数据库对象
     */
    public MiddleUser dtoToModel(MiddleUserDTO dto) {
        if (null == dto) {
            return null;
        }
        final MiddleUser model = new MiddleUser();
        model.setRId(dto.getRId());
        model.setUserId(dto.getUserId());
        model.setMobile(dto.getMobile());
        model.setUnionId(dto.getUnionId());
        model.setNickName(dto.getNickName());
        model.setOpenid(dto.getOpenid());
        model.setBindTime(dto.getBindTime());
        model.setHeadImg(dto.getHeadImg());
        model.setAppId(dto.getAppId());
        return model;
    }

    /**
     * 数据库对象转页面对象
     *
     * @param list 数据库对象
     * @return 页面对象
     */
    public List<MiddleUserDTO> modelToDtoList(List<MiddleUser> list) {
        return Optional.ofNullable(list).orElse(Collections.emptyList()).stream().map(this::modelToDto).collect(toList());
    }

    /**
     * 页面对象转数据库对象
     *
     * @param list 页面对象
     * @return 数据库对象
     */
    public List<MiddleUser> dtoToModelList(List<MiddleUserDTO> list) {
        return Optional.ofNullable(list).orElse(Collections.emptyList()).stream().map(this::dtoToModel).collect(toList());
    }

    /**
     * 拼装查询参数，除了主键的其他字段会自动加入进来
     *
     * @param dto 查询参数，不为空的属性会被自动加入
     */
    public QueryWrapper<MiddleUser> createQueryWrapper(MiddleUserDTO dto) {
        return createQueryWrapper(dto, "");
    }

    /**
     * 拼装查询参数，除了主键的其他字段会自动加入进来
     *
     * @param dto        查询参数，不为空的属性会被自动加入
     * @param tableAlias 表别名
     */
    public QueryWrapper<MiddleUser> createQueryWrapper(MiddleUserDTO dto, String tableAlias) {
        final QueryWrapper<MiddleUser> queryWrapper = new QueryWrapper<>();
        if (null != dto) {
            if (StringUtils.hasLength(tableAlias) && tableAlias.charAt(tableAlias.length() - 1) != '.') {
                tableAlias += ".";
            }
            if (StringUtils.hasLength(dto.getUserId())) {
                queryWrapper.eq(tableAlias + Constants.USER_ID, dto.getUserId());
            }
            if (StringUtils.hasLength(dto.getMobile())) {
                queryWrapper.eq(tableAlias + "mobile", dto.getMobile());
            }
            if (StringUtils.hasLength(dto.getUnionId())) {
                queryWrapper.eq(tableAlias + "union_id", dto.getUnionId());
            }
            if (StringUtils.hasLength(dto.getOpenid())) {
                queryWrapper.eq(tableAlias + Constants.OPENID, dto.getOpenid());
            }
            if (StringUtils.hasLength(dto.getAppId())) {
                queryWrapper.eq(tableAlias + Constants.APPID, dto.getAppId());
            }
            if (StringUtils.hasLength(dto.getBeginTime())) {
                queryWrapper.apply("date_format(" + tableAlias + "bind_time,'%Y-%m-%d') >= {0}", dto.getBeginTime());
            }
            if (StringUtils.hasLength(dto.getEndTime())) {
                queryWrapper.apply("date_format(" + tableAlias + "bind_time,'%Y-%m-%d') <= {0}", dto.getEndTime());
            }
        }
        queryWrapper.orderByDesc(tableAlias + "bind_time");
        return queryWrapper;
    }

    @Transactional(readOnly = true)
    public List<MiddleUser> getListByUnionId(String unionId) {
        final QueryWrapper<MiddleUser> queryWrapper = new QueryWrapper<>();
        queryWrapper.eq("union_id", unionId);
        List<MiddleUser> list = getBaseMapper().selectList(queryWrapper);
        queryWrapper.clear();
        return list;
    }
}